CUDA_VISIBLE_DEVICES=0 python2 train.py \
--dataset_name pubmed-20k \
--dir_output pubmed-20k_1 